package controllers

import beego "github.com/beego/beego/v2/server/web"

func OkResponse(r beego.Controller, data interface{}, msg string) {
	if msg == "" {
		msg = "success"
	}
	jsonData := map[string]interface{}{"code": 0, "data": data, "msg": msg}
	r.Data["json"] = jsonData
	r.ServeJSON()
	return
}

func FailResponse(r beego.Controller, msg string) {
	if msg == "" {
		msg = "fail"
	}
	jsonData := map[string]interface{}{"code": -1, "data": nil, "msg": msg}
	r.Data["json"] = jsonData
	r.ServeJSON()
}

func Response(c beego.Controller, data interface{}, err error) {
	msg := ""
	code := 0
	if err != nil {
		msg = err.Error()
		code = -1
	}
	c.Data["json"] = map[string]interface{}{"code": code, "data": data, "msg": msg}
	c.ServeJSON()
	return
}
